package me.sdau.spider.kafkaprocess.kafka.nlp.sdk;

import com.alibaba.fastjson.JSONObject;
import com.bonc.text.sdk.client.TextKeywordClient;

public class KeyWords implements NlpMethod {
    private TextKeywordClient instance = TextKeywordClient.getInstance();

    @Override
    public String get(String title, String content) {
        JSONObject jsonObject = JSONObject.parseObject(getJson(title, content));
        return jsonObject.getString("keywords");
    }

    @Override
    public String getJson(String title, String content) {
        instance.setDomain("172.16.23.13:9187");
        return instance.getKeywordsJ(content, title, numOfSub, percent, numOfAbs);
    }

    public static void main(String[] args) {
        String title = "发扬前人栽树后人乘凉精神 多种树种好树管好树";
        String content = "新华社北京4月5日电 中共中央总书记、国家主席、中央军委主席习近平5日上午在参加首都义务植树活动时强调，"
                + "中华民族伟大复兴要靠全体中华儿女共同奋斗。“十三五”时期既是全面建成小康社会的决胜阶段，也是生态文明建设的重要时期。"
                + "发展林业是全面建成小康社会的重要内容，是生态文明建设的重要举措。各级领导干部要带头参加义务植树，身体力行在全社会宣传新发展理念，"
                + "发扬前人栽树、后人乘凉精神，多种树、种好树、管好树，让大地山川绿起来，让人民群众生活环境美起来。";
        int numOfSub = 10;    //关键词个数
        int percent = 5;    //摘要占文章长度的百分比
        int numOfAbs = 3;    //摘要句数
        String str = new KeyWords().getJson(title, content);
        System.out.println(str);
    }
}
